Square Pulse Wave Graph in Python